op {
  name: "EnsureShape"
  input_arg {
    name: "input"
    type_attr: "T"
  }
  output_arg {
    name: "output"
    type_attr: "T"
  }
  attr {
    name: "shape"
    type: "shape"
  }
  attr {
    name: "T"
    type: "type"
  }
}
